package com.spzx.product.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.spzx.product.api.vo.SkuLockVo;
import com.spzx.product.pojo.SkuStock;
import com.spzx.product.mapper.SkuStockMapper;
import com.spzx.product.service.ISkuStockService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.spzx.product.vo.SkuStockVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * <p>
 * 商品sku库存表 服务实现类
 * </p>
 *
 * @author atguigu
 * @since 2025-03-12
 */
@Service
public class SkuStockServiceImpl extends ServiceImpl<SkuStockMapper, SkuStock> implements ISkuStockService {

    @Autowired
    private RedisTemplate redisTemplate;

    @Override
    public SkuStockVo getSkuStock(Long skuId) {

        // 根据skuId查询商品库存
        LambdaQueryWrapper<SkuStock> queryWrapper = new LambdaQueryWrapper<>();
        /*
        queryWrapper
                .eq(SkuStock::getSkuId, skuId)
                .select(SkuStock::getAvailableNum, SkuStock::getSaleNum);
        */
        queryWrapper.eq(SkuStock::getSkuId, skuId);
        queryWrapper.select(SkuStock::getAvailableNum, SkuStock::getSaleNum);
        SkuStock skuStock = baseMapper.selectOne(queryWrapper);

        // 封装数据
        SkuStockVo skuStockVo = new SkuStockVo();
        skuStockVo.setAvailableNum(skuStock.getAvailableNum());
        skuStockVo.setSaleNum(skuStock.getSaleNum());
        skuStockVo.setSkuId(skuId);
        return skuStockVo;
    }

    @Override
    public String checkAndLock(String orderNo, List<SkuLockVo> skuLockVoList) {

        for (SkuLockVo skuLockVo : skuLockVoList) {
            SkuStock skuStock = baseMapper.check(skuLockVo.getSkuId(), skuLockVo.getSkuNum());

            if (skuStock == null) {
                skuLockVo.setHaveStock(false);  // 无库存
            } else {
                skuLockVo.setHaveStock(true);  // 有库存
            }
        }

        // 判断各商品库存是否足够
        if (skuLockVoList.stream().anyMatch(skuLockVo -> !skuLockVo.isHaveStock())) {
            // 过滤出所有库存不足的商品
            List<SkuLockVo> list = skuLockVoList
                    .stream()
                    .filter(skuLockVo -> !skuLockVo.isHaveStock())
                    .toList();

            StringBuffer resultMsg = new StringBuffer();
            resultMsg.append("以下商品库存不足：");
            for (SkuLockVo skuLockVo : list) {
                resultMsg
                        .append("\n")
                        .append(skuLockVo.getSkuName())
                        .append("库存不足，剩余库存：")
                        .append(skuLockVo.getSkuNum())
                        .append("；");
            }

            return resultMsg.toString();
        } else {
            for (SkuLockVo skuLockVo : skuLockVoList) {
                // 锁定库存
                baseMapper.lock(skuLockVo.getSkuId(), skuLockVo.getSkuNum());
            }

            // 如果所有商品都锁定成功的情况下，需要缓存锁定信息到redis。以方便将来解锁库存或者减库存
            String dataKey = "sku:lock:data:" + orderNo;
            redisTemplate.opsForValue().set(dataKey, skuLockVoList);

            return "";
        }
    }

    @Override
    public void unlock(String orderNo) {

        // 幂等性处理
        String key = "sku:unlock:" + orderNo;
        Boolean aBoolean = redisTemplate.opsForValue().setIfAbsent(key, orderNo, 30, TimeUnit.SECONDS);
        if (!aBoolean) {
            return;
        }

        // 获取 redis中的锁定信息
        String dataKey = "sku:lock:data:" + orderNo;
        List<SkuLockVo> skuLockVoList = (List<SkuLockVo>) redisTemplate.opsForValue().get(dataKey);
        if (!CollectionUtils.isEmpty(skuLockVoList)) {
            for (SkuLockVo skuLockVo : skuLockVoList) {
                baseMapper.unlock(skuLockVo.getSkuId(), skuLockVo.getSkuNum());
            }
        }

        // 删除 redis中的锁定信息
        redisTemplate.delete(dataKey);
    }
}
